home *** CD-ROM | disk | FTP | other *** search
/ Night Owl 8 / Night Owl CD-ROM (NOPV8) (Night Owl Publisher) (1993).ISO / 017a / binutils.arj / AR.C < prev    next >
C/C++ Source or Header  |  1992-03-02  |  49KB  |  2,012 lines
  1. /* ar.c - Archive modify and extract.
  2.    Copyright (C) 1988 Free Software Foundation, Inc.
  3.  
  4.    This program is free software; you can redistribute it and/or modify
  5.    it under the terms of the GNU General Public License as published by
  6.    the Free Software Foundation; either version 1, or (at your option)
  7.    any later version.
  8.  
  9.    This program is distributed in the hope that it will be useful,
  10.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.    GNU General Public License for more details.
  13.  
  14.    You should have received a copy of the GNU General Public License
  15.    along with this program; if not, write to the Free Software
  16.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17.  
  18. #include <stdio.h>
  19. #include <ar.h>
  20. #include <errno.h>
  21. #include <sys/types.h>
  22. #include <sys/stat.h>
  23.  
  24. #if !defined(A_OUT) && !defined(MACH_O)
  25. #define A_OUT
  26. #endif
  27.  
  28. #ifdef A_OUT
  29. #ifdef COFF_ENCAPSULATE
  30. #ifdef GNUDOS
  31. #include "aoutencap.h"
  32. #else
  33. #include "a.out.encap.h"
  34. #endif
  35. #else
  36. #ifdef GNUDOS
  37. #include <aout.h>
  38. #else
  39. #include <a.out.h>
  40. #endif
  41. #endif
  42. #endif
  43.  
  44. #ifdef MACH_O
  45. #ifndef A_OUT
  46. #include <nlist.h>
  47. #endif
  48. #include <sys/loader.h>
  49. #endif
  50.  
  51. #ifdef USG
  52. #include <time.h>
  53. #include <fcntl.h>
  54. #else
  55. #include <sys/file.h>
  56. #include <sys/time.h>
  57. #endif
  58.  
  59. #ifdef    __GNUC__
  60. #define    alloca    __builtin_alloca
  61. #else
  62. # ifdef sparc
  63. #  include <alloca.h>
  64. # else
  65. char *alloca ();
  66. # endif
  67. #endif
  68.  
  69. #ifdef    USG
  70. #define    bcopy(source, dest, size)    memcpy((dest), (source), (size))
  71. #define    bcmp(a, b, size)        memcmp((a), (b), (size))
  72. #define    bzero(s, size)            memset((s), 0, (size))
  73. #endif
  74.  
  75. /* Locking is normally disabled because fcntl hangs on the Sun
  76.    and it isn't supported properly across NFS anyway.  */
  77. #ifdef LOCKS
  78. /* You might need to compile with -I/usr/include/sys if your fcntl.h
  79.    isn't in /usr/include (which is where it should be according to POSIX).  */
  80. #include <fcntl.h>
  81. #endif
  82.  
  83. /* This structure is used internally to represent the info
  84.    on a member of an archive.  This is to make it easier to change format.  */
  85.  
  86. struct member_desc
  87.   {
  88.     /* Name of member.  */
  89.     char *name;
  90.  
  91.     /* The following fields are stored in the member header as decimal or octal
  92.        numerals, but in this structure they are stored as machine numbers.  */
  93.     int mode;        /* Protection mode from member header.  */
  94.     long int date;    /* Last modify date as stored in member header.  */
  95.     unsigned int size;    /* Bytes of member's data, from member header.  */
  96.     int uid, gid;    /* UID and GID fields copied from member header.  */
  97.     unsigned int offset;/* Offset in archive of the header of this member.  */
  98.     unsigned int data_offset;/* Offset of first data byte of the member.  */
  99.  
  100.     /* The next field does not describe where the member was in the
  101.        old archive, but rather where it will be in the modified archive.
  102.        It is set up by write_archive.  */
  103.     unsigned int new_offset;    /* Offset of this member in new archive */
  104.  
  105.     /* Symdef data for member.  Used only for files being inserted.  */
  106.     struct symdef *symdefs;
  107.     unsigned int nsymdefs;    /* Number of entries of symdef data.  */
  108.     unsigned int string_size;    /* Size of strings needed by symdef data.  */
  109.   };
  110.  
  111. /* Each symbol is recorded by something like this.  */
  112.  
  113. struct symdef
  114.   {
  115.     union
  116.       {
  117.     unsigned long int stringoffset;
  118.     char *name;
  119.       } s;
  120.     unsigned long int offset;
  121.   };
  122.  
  123. /* Nonzero means it's the name of an existing member;
  124.    position new or moved files with respect to this one.  */
  125.  
  126. char *posname;
  127.  
  128.  
  129. /* How to use `posname':
  130.    POS_BEFORE means position before that member.
  131.    POS_AFTER means position after that member.
  132.    POS_DEFAULT if position by default; then `posname' should also be zero. */
  133.  
  134. enum { POS_DEFAULT, POS_BEFORE, POS_AFTER } postype;
  135.  
  136. /* Nonzero means describe each action performed.  */
  137.  
  138. int verbose;
  139.  
  140. /* Nonzero means don't warn about creating the archive file if necessary.  */
  141.  
  142. int silent_create;
  143.  
  144. /* Nonzero means don't replace existing members whose
  145.    dates are more recent than the corresponding files.  */
  146.  
  147. int newer_only;
  148.  
  149. /* Nonzero means preserve dates of members when extracting them.  */
  150.  
  151. int preserve_dates;
  152.  
  153. /* Operation to be performed.  */
  154.  
  155. #define DELETE 1
  156. #define REPLACE 2
  157. #define PRINT_TABLE 3
  158. #define PRINT_FILES 4
  159. #define EXTRACT 5
  160. #define MOVE 6
  161. #define QUICK_APPEND 7
  162.  
  163. int operation;
  164.  
  165. /* Name of archive file.  */
  166.  
  167. char *archive;
  168.  
  169. /* Descriptor on which we have locked the original archive file,
  170.    or -1 if this has not been done.  */
  171.  
  172. int lock_indesc;
  173.  
  174. /* Pointer to tail of `argv', at first subfile name argument,
  175.  or zero if no such were specified.  */
  176.  
  177. char **files;
  178.  
  179. /* Nonzero means write a __.SYMDEF member into the modified archive.  */
  180.  
  181. int symdef_flag;
  182.  
  183. /* Nonzero means __.SYMDEF member exists in old archive.  */
  184.  
  185. int symdef_exists;
  186.  
  187. /* Nonzero means don't update __.SYMDEF unless the flag was given.  */
  188.  
  189. int ignore_symdef;
  190.  
  191. /* Total number of symdef entries we will have. */
  192.  
  193. unsigned long int nsymdefs;
  194.  
  195. /* Symdef data from old archive (set up only if we need it) */
  196.  
  197. struct symdef *old_symdefs;
  198.  
  199. /* Number of symdefs in remaining in old_symdefs.  */
  200.  
  201. unsigned int num_old_symdefs;
  202.  
  203. /* Number of symdefs old_symdefs had when it was read in.  */
  204.  
  205. unsigned long int original_num_symdefs;
  206.  
  207. /* String table from old __.SYMDEF member.  */
  208.  
  209. char *old_strings;
  210.  
  211. /* Size of old_strings */
  212.  
  213. unsigned long int old_strings_size;
  214.  
  215. /* String table to be written into __.SYMDEF member.  */
  216.  
  217. char *new_strings;
  218.  
  219. /* Size of new_strings */
  220.  
  221. unsigned long int new_strings_size;
  222.  
  223. /* An archive map is a chain of these structures.
  224.   Each structure describes one member of the archive.
  225.   The chain is in the same order as the members are.  */
  226.  
  227. struct mapelt
  228.   {
  229.     struct member_desc info;
  230.     struct mapelt *next;
  231.   };
  232.  
  233. struct mapelt *maplast;
  234.  
  235. /* If nonzero, this is the map-element for the __.SYMDEF member
  236.    and we should update the time of that member just before finishing.  */
  237.  
  238. struct mapelt *symdef_mapelt;
  239.  
  240. /* Header that we wrote for the __.SYMDEF member.  */
  241.  
  242. struct ar_hdr symdef_header;
  243.  
  244. char *xmalloc (), *xrealloc ();
  245. void free ();
  246.  
  247. void add_to_map (), delete_from_map ();
  248. int insert_in_map ();
  249. void print_descr ();
  250. char *concat ();
  251. void scan ();
  252. void extract_members ();
  253. void extract_member ();
  254. void print_contents ();
  255. void write_symdef_member ();
  256. void read_old_symdefs ();
  257. void two_operations ();
  258. void usage (), fatal (), error (), error_with_file ();
  259. void perror_with_name (), pfatal_with_name ();
  260. void write_archive ();
  261. void touch_symdef_member ();
  262. void update_symdefs ();
  263. void delete_members (), move_members (), replace_members ();
  264. void quick_append ();
  265.  
  266. /* Output BYTES of data at BUF to the descriptor DESC.
  267.    FILE is the name of the file (for error messages).  */
  268.  
  269. void
  270. mywrite (desc, buf, bytes, file)
  271.      int desc;
  272.      char *buf;
  273.      int bytes;
  274.      char *file;
  275. {
  276.   register int val;
  277.  
  278.   while (bytes > 0)
  279.     {
  280.       val = write (desc, buf, bytes);
  281.       if (val <= 0)
  282.     perror_with_name (file);
  283.       buf += val;
  284.       bytes -= val;
  285.     }
  286. }
  287.  
  288. int
  289. main (argc, argv)
  290.      int argc;
  291.      char **argv;
  292. {
  293.   int i;
  294.  
  295. #ifdef GNUDOS
  296.   _fmode = O_BINARY; /* set default file type */
  297. #endif
  298.  
  299.   operation = 0;
  300.   verbose = 0;
  301.   newer_only = 0;
  302.   silent_create = 0;
  303.   posname = 0;
  304.   postype = POS_DEFAULT;
  305.   preserve_dates = 0;
  306.   symdef_flag = 0;
  307.   symdef_exists = 0;
  308.   ignore_symdef = 0;
  309.   symdef_mapelt = 0;
  310.   files = 0;
  311.   lock_indesc = -1;
  312.  
  313.   if (argc < 2)
  314.     usage ("too few command arguments", 0);
  315.  
  316.   {
  317.     char *key = argv[1];
  318.     char *p = key;
  319.     char c;
  320.  
  321.     while (c = *p++)
  322.       {
  323.     switch (c)
  324.       {
  325.       case 'a':
  326.         postype = POS_AFTER;
  327.         break;
  328.  
  329.       case 'b':
  330.         postype = POS_BEFORE;
  331.         break;
  332.  
  333.       case 'c':
  334.         silent_create = 1;
  335.         break;
  336.  
  337.       case 'd':
  338.         if (operation)
  339.           two_operations ();
  340.  
  341.         operation = DELETE;
  342.         break;
  343.  
  344.       case 'i':
  345.         postype = POS_BEFORE;
  346.         break;
  347.  
  348.       case 'l':
  349.         break;
  350.  
  351.       case 'm':
  352.         if (operation)
  353.           two_operations ();
  354.         operation = MOVE;
  355.         break;
  356.  
  357.       case 'o':
  358.         preserve_dates = 1;
  359.         break;
  360.  
  361.       case 'p':
  362.         if (operation)
  363.           two_operations ();
  364.         operation = PRINT_FILES;
  365.         break;
  366.  
  367.       case 'q':
  368.         if (operation)
  369.           two_operations ();
  370.         operation = QUICK_APPEND;
  371.         break;
  372.  
  373.       case 'r':
  374.         if (operation)
  375.           two_operations ();
  376.         operation = REPLACE;
  377.         break;
  378.  
  379.       case 's':
  380.         symdef_flag = 1;
  381.         break;
  382.  
  383.       case 't':
  384.         if (operation)
  385.           two_operations ();
  386.         operation = PRINT_TABLE;
  387.         break;
  388.  
  389.       case 'u':
  390.         operation = REPLACE;
  391.         newer_only = 1;
  392.         break;
  393.  
  394.       case 'v':
  395.         verbose = 1;
  396.         break;
  397.  
  398.       case 'x':
  399.         if (operation)
  400.           two_operations ();
  401.         operation = EXTRACT;
  402.         break;
  403.       }
  404.       }
  405.   
  406.   }
  407.  
  408.   if (operation == 0 && symdef_flag)
  409.     operation = REPLACE;
  410.  
  411.   if (operation == 0)
  412.     usage ("no operation specified", 0);
  413.  
  414.   i = 2;
  415.  
  416.   if (postype != POS_DEFAULT)
  417.     posname = argv[i++];
  418.  
  419.   archive = argv[i++];
  420.  
  421.   if (i < argc)
  422.     {
  423.       files = &argv[i];
  424.       while (i < argc)
  425.     if (!strcmp (argv[i++], "__.SYMDEF"))
  426.       {
  427.         ignore_symdef = 1;
  428.         break;
  429.       }
  430.     }
  431.  
  432.   switch (operation)
  433.     {
  434.     case EXTRACT:
  435.     extract_members (extract_member);
  436.     break;
  437.  
  438.     case PRINT_TABLE:
  439.     extract_members (print_descr);
  440.     break;
  441.  
  442.     case PRINT_FILES:
  443.     extract_members (print_contents);
  444.     break;
  445.  
  446.     case DELETE:
  447.     if (files != 0)
  448.       delete_members ();
  449.     break;
  450.  
  451.     case MOVE:
  452.     if (files != 0)
  453.       move_members ();
  454.     break;
  455.  
  456.     case REPLACE:
  457.     if (files != 0 || symdef_flag)
  458.       replace_members ();
  459.     break;
  460.  
  461.     case QUICK_APPEND:
  462.     if (files != 0)
  463.       quick_append ();
  464.     break;
  465.  
  466.     default:
  467.     usage ("invalid operation %d", operation);
  468.     }
  469.  
  470.   exit (0);
  471.   return 0;
  472. }
  473.  
  474. void
  475. two_operations ()
  476. {
  477.   usage ("two different operation switches specified", 0);
  478. }
  479.  
  480. void
  481. scan (function, crflag)
  482.      void (*function) ();
  483.      int crflag;
  484. {
  485.   FILE *arcstream = fopen (archive, "r");
  486.  
  487.   if (arcstream == 0 && crflag)
  488.     /* Creation-warning, if desired, will happen later.  */
  489.     return;
  490.  
  491.   if (arcstream == 0)
  492.     {
  493.       perror_with_name (archive);
  494.       exit (1);
  495.     }
  496.   {
  497.     char buf[SARMAG];
  498.     int nread = fread (buf, 1, SARMAG, arcstream);
  499.     if (nread != SARMAG || bcmp (buf, ARMAG, SARMAG))
  500.       fatal ("file %s not a valid archive", archive);
  501.   }
  502.  
  503.   /* Now find the members one by one.  */
  504.   {
  505.     int member_offset = SARMAG;
  506.     while (1)
  507.       {
  508.     int nread;
  509.     struct ar_hdr member_header;
  510.     struct member_desc member_desc;
  511.     char name [1 + sizeof member_header.ar_name];
  512.  
  513.     if (fseek (arcstream, member_offset, 0) < 0)
  514.       perror_with_name (archive);
  515.  
  516.     nread = fread (&member_header, 1, sizeof (struct ar_hdr), arcstream);
  517.     if (nread == 0)
  518.       /* No data left means end of file; that is OK.  */
  519.       break;
  520.  
  521.     if (nread != sizeof (member_header)
  522.         || bcmp (member_header.ar_fmag, ARFMAG, 2))
  523.       fatal ("file %s not a valid archive", archive);
  524.     bcopy (member_header.ar_name, name, sizeof member_header.ar_name);
  525.     {
  526.       char *p = name + sizeof member_header.ar_name;
  527.       *p = '\0';
  528.       while (p > name && *--p == ' ')
  529.         *p = '\0';
  530.     }
  531.     member_desc.name = name;
  532.     sscanf (member_header.ar_mode, "%o", &member_desc.mode);
  533.     member_desc.date = atoi (member_header.ar_date);
  534.     member_desc.size = atoi (member_header.ar_size);
  535.     member_desc.uid = atoi (member_header.ar_uid);
  536.     member_desc.gid = atoi (member_header.ar_gid);
  537.     member_desc.offset = member_offset;
  538.     member_desc.data_offset = member_offset + sizeof (member_header);
  539.  
  540.     member_desc.new_offset = 0;
  541.     member_desc.symdefs = 0;
  542.     member_desc.nsymdefs = 0;
  543.     member_desc.string_size = 0;
  544.  
  545.     if (!ignore_symdef && !strcmp (name, "__.SYMDEF"))
  546.       symdef_exists = 1;
  547.  
  548.     function (member_desc, arcstream);
  549.  
  550.     member_offset += sizeof (member_header) + member_desc.size;
  551.     if (member_offset & 1)
  552.       ++member_offset;
  553.       }
  554.   }
  555.  
  556.   
  557.   fclose (arcstream);
  558. }
  559.  
  560. void print_modes ();
  561.  
  562. void
  563. print_descr (member)
  564.      struct member_desc member;
  565. {
  566.   char *timestring;
  567.   if (!verbose)
  568.     {
  569.       puts (member.name);
  570.       return;
  571.     }
  572.   print_modes (member.mode);
  573.   timestring = ctime (&member.date);
  574.   printf (" %2d/%2d %6d %12.12s %4.4s %s\n",
  575.       member.uid, member.gid,
  576.       member.size, timestring + 4, timestring + 20,
  577.       member.name);
  578. }
  579.  
  580. void
  581. print_modes (modes)
  582.      int modes;
  583. {
  584.   putchar (modes & 0400 ? 'r' : '-');
  585.   putchar (modes & 0200 ? 'w' : '-');
  586.   putchar (modes & 0100 ? 'x' : '-');
  587.   putchar (modes & 040 ? 'r' : '-');
  588.   putchar (modes & 020 ? 'w' : '-');
  589.   putchar (modes & 010 ? 'x' : '-');
  590.   putchar (modes & 04 ? 'r' : '-');
  591.   putchar (modes & 02 ? 'w' : '-');
  592.   putchar (modes & 01 ? 'x' : '-');
  593. }
  594.  
  595. #define BUFSIZE 1024
  596.  
  597. void
  598. extract_member (member, istream)
  599.      struct member_desc member;
  600.      FILE *istream;
  601. {
  602.   int ncopied = 0;
  603.   FILE *ostream;
  604.  
  605.   fseek (istream, member.data_offset, 0);
  606.   ostream = fopen (member.name, "w");
  607.   if (!ostream)
  608.     {
  609.       perror_with_name (member.name);
  610.       return;
  611.     }
  612.  
  613.   if (verbose)
  614.     printf ("x - %s\n", member.name);
  615.  
  616.   while (ncopied < member.size)
  617.     {
  618.       char buf [BUFSIZE];
  619.       int tocopy = member.size - ncopied;
  620.       int nread;
  621.       if (tocopy > BUFSIZE) tocopy = BUFSIZE;
  622.       nread = fread (buf, 1, tocopy, istream);
  623.       if (nread != tocopy)
  624.     fatal ("file %s not a valid archive", archive);
  625.       fwrite (buf, 1, nread, ostream);
  626.       ncopied += tocopy;
  627.     }
  628.  
  629. #ifdef USG
  630.   chmod (member.name, member.mode);
  631. #else
  632.   fchmod (fileno (ostream), member.mode);
  633. #endif
  634.   if (ferror (ostream) || fclose (ostream) != 0)
  635.     error ("%s: I/O error", member.name);
  636.  
  637.   if (preserve_dates)
  638.     {
  639. #ifdef USG
  640.       long tv[2];
  641.       tv[0] = member.date;
  642.       tv[1] = member.date;
  643.       utime (member.name, tv);
  644. #else
  645.       struct timeval tv[2];
  646.       tv[0].tv_sec = member.date;
  647.       tv[0].tv_usec = 0;
  648.       tv[1].tv_sec = member.date;
  649.       tv[1].tv_usec = 0;
  650.       utimes (member.name, tv);
  651. #endif
  652.     }
  653. }
  654.  
  655. void
  656. print_contents (member, istream)
  657.      struct member_desc member;
  658.      FILE *istream;
  659. {
  660.   int ncopied = 0;
  661.  
  662.   fseek (istream, member.data_offset, 0);
  663.  
  664.   if (verbose)
  665.   printf ("\n<member %s>\n\n", member.name);
  666.  
  667.   while (ncopied < member.size)
  668.     {
  669.       char buf [BUFSIZE];
  670.       int tocopy = member.size - ncopied;
  671.       int nread;
  672.       if (tocopy > BUFSIZE) tocopy = BUFSIZE;
  673.       nread = fread (buf, 1, tocopy, istream);
  674.       if (nread != tocopy)
  675.     fatal ("file %s not a valid archive", archive);
  676.       fwrite (buf, 1, nread, stdout);
  677.       ncopied += tocopy;
  678.     }
  679. }
  680.  
  681. /* Make a map of the existing members of the archive: their names,
  682.  positions and sizes.  */
  683.  
  684. /* If `nonexistent_ok' is nonzero,
  685.  just return 0 for an archive that does not exist.
  686.  This will cause the ordinary supersede procedure to
  687.  create a new archive.  */
  688.  
  689. struct mapelt *
  690. make_map (nonexistent_ok)
  691.      int nonexistent_ok;
  692. {
  693.   struct mapelt mapstart;
  694.   mapstart.next = 0;
  695.   maplast = &mapstart;
  696.   scan (add_to_map, nonexistent_ok);
  697.   return mapstart.next;
  698. }
  699.  
  700. void
  701. add_to_map (member)
  702.      struct member_desc member;
  703. {
  704.   struct mapelt *mapelt = (struct mapelt *) xmalloc (sizeof (struct mapelt));
  705.   mapelt->info = member;
  706.   mapelt->info.name = concat (mapelt->info.name, "", "");
  707.   maplast->next = mapelt;
  708.   mapelt->next = 0;
  709.   maplast = mapelt;
  710. }
  711.  
  712. /* Return the last element of the specified map.  */
  713.  
  714. struct mapelt *
  715. last_mapelt (map)
  716.      struct mapelt *map;
  717. {
  718.   struct mapelt *tail = map;
  719.   while (tail->next) tail = tail->next;
  720.   return tail;
  721. }
  722.  
  723. /* Return the element of the specified map which precedes elt.  */
  724.  
  725. struct mapelt *
  726. prev_mapelt (map, elt)
  727.      struct mapelt *map, *elt;
  728. {
  729.   struct mapelt *tail = map;
  730.   while (tail->next && tail->next != elt)
  731.     tail = tail->next;
  732.   if (tail->next) return tail;
  733.   return 0;
  734. }
  735.  
  736. /* Return the element of the specified map which has the specified name.  */
  737.  
  738. struct mapelt *
  739. find_mapelt_noerror (map, name)
  740.      struct mapelt *map;
  741.      register char *name;
  742. {
  743.   register struct mapelt *tail;
  744.   unsigned int len = strlen (name);
  745.   int dot_o = name[len - 2] == '.' && name[len - 1] == 'o';
  746.  
  747.   for (tail = map; tail != 0; tail = tail->next)
  748.     {
  749.       if (tail->info.name == 0)
  750.     continue;
  751.       if (!strncmp (tail->info.name, name, 13))
  752.     {
  753.       unsigned int eltlen = strlen (tail->info.name);
  754.       if (len <= 13 || eltlen <= 13)
  755.         return tail;
  756.       else
  757.         {
  758.           char *p = tail->info.name + 13;
  759.           if (dot_o && p[0] == '.' && p[1] == 'o' && p[2] == '\0')
  760.         return tail;
  761.           else if (!strncmp (p, name + 13,
  762.                  (len > eltlen ? len : eltlen) - 13))
  763.         return tail;
  764.         }
  765.     }
  766.     }
  767.  
  768.   return 0;
  769. }
  770.  
  771. struct mapelt *
  772. find_mapelt (map, name)
  773.      struct mapelt *map;
  774.      char *name;
  775. {
  776.   register struct mapelt *found = find_mapelt_noerror (map, name);
  777.   if (found == 0)
  778.     error ("no member named `%s'", name);
  779.   return found;
  780. }
  781.  
  782. /* Before looking at the archive, if we are going to update it
  783.    based on looking at its current contents, make an exclusive lock on it.
  784.    The lock is released when `write_archive' is called.  */
  785.  
  786. void
  787. lock_for_update ()
  788. {
  789.   /* Open the existing archive file; if that fails, create an empty one.  */
  790.  
  791.   lock_indesc = open (archive, O_RDWR, 0);
  792.  
  793.   if (lock_indesc < 0)
  794.     {
  795.       int outdesc;
  796.  
  797.       if (!silent_create)
  798.     printf ("Creating archive file `%s'\n", archive);
  799.       outdesc = open (archive, O_WRONLY | O_APPEND | O_CREAT, 0666);
  800.       if (outdesc < 0)
  801.     pfatal_with_name (archive);
  802.       mywrite (outdesc, ARMAG, SARMAG, archive);
  803.       close (outdesc);
  804.  
  805.       /* Now we had better be able to open for update!  */
  806.  
  807.       lock_indesc = open (archive, O_RDWR, 0);
  808.       if (lock_indesc < 0)
  809.     pfatal_with_name (archive);
  810.     }
  811.  
  812. #ifdef LOCKS
  813.   /* Lock the old file so that it won't be updated by two programs at once.
  814.      This uses the fcntl locking facility found on Sun systems
  815.      which is also in POSIX.  (Perhaps it comes from sysV.)
  816.  
  817.      Note that merely reading an archive does not require a lock,
  818.      because we use `rename' to update the whole file atomically.  */
  819.  
  820.   {
  821.     struct flock lock;
  822.  
  823.     lock.l_type = F_WRLCK;
  824.     lock.l_whence = 0;
  825.     lock.l_start = 0;
  826.     lock.l_len = 0;
  827.  
  828.     while (1)
  829.       {
  830.     int value = fcntl (lock_indesc, F_SETLKW, &lock);
  831.     if (value >= 0)
  832.       break;
  833.     else if (errno == EINTR)
  834.       continue;
  835.     else
  836.       pfatal_with_name ("locking archive");
  837.       }
  838.   }
  839. #endif
  840. }
  841.  
  842. /* Unlock archive and close the file descriptor.  */
  843.  
  844. void
  845. close_archive ()
  846. {
  847. #ifdef LOCKS
  848.   {
  849.     struct flock lock;
  850.  
  851.     /* Unlock the old archive.  */
  852.  
  853.     lock.l_type = F_UNLCK;
  854.     lock.l_whence = 0;
  855.     lock.l_start = 0;
  856.     lock.l_len = 0;
  857.  
  858.     fcntl (lock_indesc, F_SETLK, &lock);
  859.   }
  860. #endif
  861.  
  862.   /* Close the archive.  If we renamed a new one, the old one disappears.  */
  863.   close (lock_indesc);
  864. }
  865.  
  866. /* Write a new archive file from a given map.  */
  867. /* When a map is used as the pattern for a new archive,
  868.  each element represents one member to put in it, and
  869.  the order of elements controls the order of writing.
  870.  
  871.  Ordinarily, the element describes a member of the old
  872.  archive, to be copied into the new one.
  873.  
  874.  If the `offset' field of the element's info is 0,
  875.  then the element describes a file to be copied into the
  876.  new archive.  The `name' field is the file's name.
  877.  
  878.  If the `name' field of an element is 0, the element is ignored.
  879.  This makes it easy to specify deletion of archive members.
  880.  
  881.  Every operation that will eventually call `write_archive'
  882.  should call `lock_for_update' before beginning
  883.  to do any I/O on the archive file.
  884. */
  885.  
  886. char *make_tempname ();
  887. void copy_out_member ();
  888.  
  889. void
  890. write_archive (map, appendflag)
  891.      struct mapelt *map;
  892.      int appendflag;
  893. {
  894.   char *tempname = make_tempname (archive);
  895.   int indesc = lock_indesc;
  896.   int outdesc;
  897.   char *outname;
  898.   struct mapelt *tail;
  899.  
  900.   /* Now open the output.  */
  901.  
  902.   if (!appendflag)
  903.     {
  904.       /* Updating an existing archive normally.
  905.      Write output as TEMPNAME and rename at the end.
  906.      There can never be two invocations trying to do this at once,
  907.      because of the lock made on the old archive file.  */
  908.  
  909.       outdesc = open (tempname, O_WRONLY | O_CREAT, 0666);
  910.       if (outdesc < 0)
  911.     pfatal_with_name (tempname);
  912.       outname = tempname;
  913.       mywrite (outdesc, ARMAG, SARMAG, outname);
  914.     }
  915.   else
  916.     {
  917.       /* Fast-append to existing archive.  */
  918.  
  919.       outdesc = open (archive, O_WRONLY | O_APPEND, 0);
  920.       if (outdesc < 0)
  921.     pfatal_with_name (archive);
  922.       outname = archive;
  923.     }
  924.  
  925.   /* If archive has or should have a __.SYMDEF member,
  926.      compute the contents for it.  */
  927.  
  928.   if (symdef_flag || symdef_exists)
  929.     {
  930.       if (symdef_exists)
  931.     read_old_symdefs (map, indesc);
  932.       else
  933.     {
  934.       struct mapelt *this = (struct mapelt *)
  935.         xmalloc (sizeof (struct mapelt));
  936.       this->info.name = "__.SYMDEF";
  937.       this->info.offset = SARMAG;
  938.       this->info.data_offset = SARMAG + sizeof (struct ar_hdr);
  939.       this->info.new_offset = 0;
  940.       this->info.date = 0;
  941.       this->info.size = 0;
  942.       this->info.uid = 0;
  943.       this->info.gid = 0;
  944.       this->info.mode = 0666;
  945.       this->info.symdefs = 0;
  946.       this->info.nsymdefs = 0;
  947.       this->info.string_size = 0;
  948.       this->next = map;
  949.       map = this;
  950.       original_num_symdefs = 0;
  951.       old_strings_size = 0;
  952.     }
  953.  
  954.       update_symdefs (map, indesc);
  955.     }
  956.  
  957.   /* Copy the members into the output, either from the old archive
  958.      or from specified files.  */
  959.  
  960.   for (tail = map; tail != 0; tail = tail->next)
  961.     {
  962.       if ((symdef_flag || symdef_exists) && tail->info.name
  963.       && !strcmp (tail->info.name, "__.SYMDEF")
  964. #if 0
  965.       && tail->info.date==0
  966. #endif
  967.       )
  968.     write_symdef_member (tail, map, outdesc, outname);
  969.       else
  970.     copy_out_member (tail, indesc, outdesc, outname);
  971.     }
  972.  
  973.   if (symdef_mapelt != 0)
  974.     {
  975.       /* Check for members whose data offsets weren't
  976.      known when the symdef member was first written.  */
  977.       int doneany = 0;
  978.       for (tail = map; tail != 0; tail = tail->next)
  979.     if (tail->info.offset == 0)
  980.       {
  981.         /* Fix up the symdefs.  */
  982.         register unsigned int i;
  983.         for (i = 0; i < tail->info.nsymdefs; ++i)
  984.           tail->info.symdefs[i].offset = tail->info.new_offset;
  985.         doneany = 1;
  986.       }
  987.       if (doneany)
  988.     {
  989.       /* Some files had bad symdefs; rewrite the symdef member.  */
  990.       lseek (outdesc, symdef_mapelt->info.offset, 0);
  991.       write_symdef_member (symdef_mapelt, map, outdesc, outname);
  992.     }
  993.     }
  994.  
  995.   /* Mark the __.SYMDEF member as up to date.  */
  996.  
  997.   if (symdef_mapelt != 0)
  998.     touch_symdef_member (outdesc, outname);
  999.  
  1000.   /* Install the new output under the intended name.  */
  1001.  
  1002. #ifndef USG
  1003.   fsync (outdesc);
  1004. #endif
  1005.   close (outdesc);
  1006.  
  1007.   if (!appendflag)
  1008.     if (rename (tempname, archive))
  1009.       pfatal_with_name (tempname);
  1010.  
  1011.   close_archive ();
  1012. }
  1013.  
  1014. void
  1015. header_from_map (header, mapelt)
  1016.      struct ar_hdr *header;
  1017.      struct mapelt *mapelt;
  1018. {
  1019.   char *nodir_name = mapelt->info.name, *cp;
  1020.   unsigned int namelen;
  1021.  
  1022.   cp = strrchr(nodir_name, '/');
  1023.   if (!cp) cp = strrchr(nodir_name, '\\');
  1024.   if (cp) nodir_name = cp+1;
  1025.  
  1026.   /* Zero the header, then store in the data as text.  */
  1027.   bzero ((char *) header, sizeof (*header));
  1028.  
  1029.   strncpy (header->ar_name, nodir_name, sizeof (header->ar_name));
  1030.   namelen = strlen (nodir_name);
  1031.   if (namelen >= sizeof (header->ar_name))
  1032.     {
  1033.       if (mapelt->info.name[namelen - 2] == '.' &&
  1034.       mapelt->info.name[namelen - 1] == 'o')
  1035.     {
  1036.       header->ar_name[sizeof (header->ar_name) - 3] = '.';
  1037.       header->ar_name[sizeof (header->ar_name) - 2] = 'o';
  1038.     }
  1039.       header->ar_name[sizeof (header->ar_name) - 1] = '\0';
  1040.       error ("member name `%s' truncated to `%s'",
  1041.          nodir_name, header->ar_name);
  1042.     }
  1043.  
  1044.   sprintf (header->ar_date, "%ld", mapelt->info.date);
  1045.   sprintf (header->ar_size, "%d", mapelt->info.size);
  1046.   sprintf (header->ar_uid, "%d", mapelt->info.uid);
  1047.   sprintf (header->ar_gid, "%d", mapelt->info.gid);
  1048.   sprintf (header->ar_mode, "%o", mapelt->info.mode);
  1049.   strncpy (header->ar_fmag, ARFMAG, sizeof (header->ar_fmag));
  1050.  
  1051.   /* Change all remaining nulls in the header into spaces.  */
  1052.   {
  1053.     char *end = (char *) &header[1];
  1054.     register char *p;
  1055.     for (p = (char *) header; p < end; ++p)
  1056.       if (*p == '\0')
  1057.     *p = ' ';
  1058.   }
  1059. }
  1060.  
  1061. /* writes to file open on OUTDESC with name OUTNAME.  */
  1062. void
  1063. copy_out_member (mapelt, archive_indesc, outdesc, outname)
  1064.      struct mapelt *mapelt;
  1065.      int archive_indesc;
  1066.      int outdesc;
  1067. {
  1068.   struct ar_hdr header;
  1069.   int indesc;
  1070.  
  1071.   if (mapelt->info.name == 0)
  1072.     /* This element was cancelled.  */
  1073.     return;
  1074.  
  1075.   header_from_map (&header, mapelt);
  1076.  
  1077.   if (mapelt->info.offset != 0)
  1078.     {
  1079.       indesc = archive_indesc;
  1080.       lseek (indesc, mapelt->info.data_offset, 0);
  1081.     }
  1082.   else
  1083.     {
  1084.       indesc = open (mapelt->info.name, 0, 0);
  1085.       if (indesc < 0)
  1086.     {
  1087.       perror_with_name (mapelt->info.name);
  1088.       return;
  1089.     }
  1090.     }
  1091.  
  1092.   mywrite (outdesc, &header, sizeof (header), outname);
  1093.  
  1094.   if (mapelt->info.data_offset == 0)
  1095.     mapelt->info.data_offset = lseek (outdesc, 0L, 1);
  1096.  
  1097.   {
  1098.     char buf[BUFSIZE];
  1099.     int tocopy = mapelt->info.size;
  1100.     while (tocopy > 0)
  1101.       {
  1102.     int thistime = tocopy;
  1103.     if (thistime > BUFSIZE) thistime = BUFSIZE;
  1104.         read (indesc, buf, thistime);
  1105.     mywrite (outdesc, buf, thistime, outname);
  1106.     tocopy -= thistime;
  1107.       }
  1108.   }
  1109.  
  1110.   if (indesc != archive_indesc)
  1111.     close (indesc);
  1112.  
  1113.   if (mapelt->info.size & 1)
  1114.     mywrite (outdesc, "\n", 1, outname);
  1115. }
  1116.  
  1117. /* Update the time of the __.SYMDEF member; done when we updated
  1118.    that member, just before we close the new archive file.
  1119.    It is open on OUTDESC and its name is OUTNAME.  */
  1120.  
  1121. void
  1122. touch_symdef_member (outdesc, outname)
  1123.      int outdesc;
  1124.      char *outname;
  1125. {
  1126.   struct stat statbuf;
  1127.   int i;
  1128.  
  1129.   /* See what mtime the archive file has as a result of our writing it.  */
  1130.   fstat (outdesc, &statbuf);
  1131.  
  1132.   /* Advance member's time to that time.  */
  1133.   bzero (symdef_header.ar_date, sizeof symdef_header.ar_date);
  1134.   sprintf (symdef_header.ar_date, "%ld", statbuf.st_mtime);
  1135.   for (i = 0; i < sizeof symdef_header.ar_date; i++)
  1136.     if (symdef_header.ar_date[i] == 0)
  1137.       symdef_header.ar_date[i] = ' ';
  1138.  
  1139.   /* Write back this member's header with the new time.  */
  1140.   if (lseek (outdesc, symdef_mapelt->info.new_offset, 0) >= 0)
  1141.     mywrite (outdesc, &symdef_header, sizeof symdef_header, outname);
  1142. }
  1143.  
  1144. char *
  1145. make_tempname (name)
  1146.      char *name;
  1147. {
  1148.   return concat (name, "", "_supersede");
  1149. }
  1150.  
  1151. void
  1152. delete_members ()
  1153. {
  1154.   struct mapelt *map = make_map (0);
  1155.   struct mapelt *tail;
  1156.   struct mapelt mapstart;
  1157.   char **p;
  1158.  
  1159.   mapstart.info.name = 0;
  1160.   mapstart.next = map;
  1161.   map = &mapstart;
  1162.  
  1163.   lock_for_update ();
  1164.  
  1165.   if (files)
  1166.     for (p = files; *p; p++)
  1167.       {
  1168.     /* If user says to delete the __.SYMDEF member,
  1169.        don't make a new one to replace it.  */
  1170.     if (!strcmp (*p, "__.SYMDEF"))
  1171.       symdef_exists = 0;
  1172.     delete_from_map (*p, map);
  1173.       }
  1174.  
  1175.   write_archive (map->next, 0);
  1176. }
  1177.  
  1178. void
  1179. delete_from_map (name, map)
  1180.      char *name;
  1181.      struct mapelt *map;
  1182. {
  1183.   struct mapelt *this = find_mapelt (map, name);
  1184.  
  1185.   if (!this) return;
  1186.   this->info.name = 0;
  1187.   if (verbose)
  1188.     printf ("d - %s\n", name);
  1189. }
  1190.  
  1191. void
  1192. move_members ()
  1193. {
  1194.   struct mapelt *map = make_map (0);
  1195.   char **p;
  1196.   struct mapelt *after_mapelt;
  1197.   struct mapelt mapstart;
  1198.   struct mapelt *change_map;
  1199.  
  1200.   mapstart.info.name = 0;
  1201.   mapstart.next = map;
  1202.   change_map = &mapstart;
  1203.  
  1204.   lock_for_update ();
  1205.  
  1206.   switch (postype)
  1207.     {
  1208.     case POS_DEFAULT:
  1209.       after_mapelt = last_mapelt (change_map);
  1210.       break;
  1211.  
  1212.     case POS_AFTER:
  1213.       after_mapelt = find_mapelt (map, posname);
  1214.       break;
  1215.  
  1216.     case POS_BEFORE:
  1217.       after_mapelt = prev_mapelt (change_map, find_mapelt (map, posname));
  1218.     }
  1219.  
  1220.   /* Failure to find specified "before" or "after" member
  1221.      is a fatal error; message has already been printed.  */
  1222.  
  1223.   if (!after_mapelt) exit (1);
  1224.  
  1225.   if (files)
  1226.     for (p = files; *p; p++)
  1227.       {
  1228.     if (move_in_map (*p, change_map, after_mapelt))
  1229.       after_mapelt = after_mapelt->next;
  1230.       }
  1231.  
  1232.   write_archive (map, 0);
  1233. }
  1234.  
  1235. int
  1236. move_in_map (name, map, after)
  1237.      char *name;
  1238.      struct mapelt *map, *after;
  1239. {
  1240.   struct mapelt *this = find_mapelt (map, name);
  1241.   struct mapelt *prev;
  1242.  
  1243.   if (!this) return 0;
  1244.   prev = prev_mapelt (map, this);
  1245.   prev->next = this->next;
  1246.   this->next = after->next;
  1247.   after->next = this;
  1248.   return 1;
  1249. }
  1250.  
  1251. /* Insert files into the archive.  */
  1252.  
  1253. void
  1254. replace_members ()
  1255. {
  1256.   struct mapelt *map = make_map (1);
  1257.   struct mapelt mapstart;
  1258.   struct mapelt *after_mapelt;
  1259.   struct mapelt *change_map;
  1260.   char **p;
  1261.   int changed;
  1262.  
  1263.   mapstart.info.name = 0;
  1264.   mapstart.next = map;
  1265.   change_map = &mapstart;
  1266.  
  1267.   lock_for_update ();
  1268.  
  1269.   switch (postype)
  1270.     {
  1271.     case POS_DEFAULT:
  1272.       after_mapelt = last_mapelt (change_map);
  1273.       break;
  1274.  
  1275.     case POS_AFTER:
  1276.       after_mapelt = find_mapelt (map, posname);
  1277.       break;
  1278.  
  1279.     case POS_BEFORE:
  1280.       after_mapelt = prev_mapelt (change_map, find_mapelt (map, posname));
  1281.     }
  1282.  
  1283.   /* Failure to find specified "before" or "after" member
  1284.      is a fatal error; the message has already been printed.  */
  1285.   if (after_mapelt == 0)
  1286.     exit (1);
  1287.  
  1288.   changed = 0;
  1289.   if (files != 0)
  1290.     for (p = files; *p != 0; ++p)
  1291.       if (insert_in_map (*p, change_map, after_mapelt))
  1292.     {
  1293.       after_mapelt = after_mapelt->next;
  1294.       changed = 1;
  1295.     }
  1296.  
  1297.   change_map = change_map->next;
  1298.   if (!changed && (!symdef_flag || symdef_exists))
  1299.     /* Nothing changed.  */
  1300.     close_archive (change_map);
  1301.   else
  1302.     write_archive (change_map, 0);
  1303. }
  1304.  
  1305. /* Handle the "quick insert" operation.  */
  1306.  
  1307. void
  1308. quick_append ()
  1309. {
  1310.   struct mapelt *map;
  1311.   struct mapelt *after;
  1312.   struct mapelt mapstart;
  1313.   char **p;
  1314.  
  1315.   mapstart.info.name = 0;
  1316.   mapstart.next = 0;
  1317.   map = &mapstart;
  1318.   after = map;
  1319.  
  1320.   lock_for_update ();
  1321.  
  1322.   /* Insert the specified files into the "map",
  1323.      but is a map of the inserted files only,
  1324.      and starts out empty.  */
  1325.   if (files)
  1326.     for (p = files; *p; p++)
  1327.       {
  1328.     if (insert_in_map (*p, map, after))
  1329.       after = after->next;
  1330.       }
  1331.  
  1332.   /* Append these files to the end of the existing archive file.  */
  1333.  
  1334.   write_archive (map->next, 1);
  1335. }
  1336.  
  1337. /* Insert an entry for name NAME into the map MAP after the map entry AFTER.
  1338.    Delete an old entry for NAME.
  1339.    MAP is assumed to start with a dummy entry, which facilitates
  1340.    insertion at the beginning of the list.
  1341.    Return 1 if successful, 0 if did nothing because file NAME doesn't
  1342.    exist or (optionally) is older.  */
  1343.  
  1344. int
  1345. insert_in_map (name, map, after)
  1346.      char *name;
  1347.      struct mapelt *map, *after;
  1348. {
  1349.   struct mapelt *old = find_mapelt_noerror (map, name);
  1350.   struct mapelt *this;
  1351.   struct stat status;
  1352.  
  1353.   if (stat (name, &status))
  1354.     {
  1355.       perror_with_name (name);
  1356.       return 0;
  1357.     }
  1358.   if (old && newer_only && status.st_mtime <= old->info.date)
  1359.     return 0;
  1360.   if (old)
  1361.     /* Delete the old one.  */
  1362.     old->info.name = 0;
  1363.   this = (struct mapelt *) xmalloc (sizeof (struct mapelt));
  1364.   this->info.name = name;
  1365.   this->info.offset = 0;
  1366.   this->info.data_offset = 0;
  1367.   this->info.date = status.st_mtime;
  1368.   this->info.size = status.st_size;
  1369.   this->info.uid = status.st_uid;
  1370.   this->info.gid = status.st_gid;
  1371.   this->info.mode = status.st_mode;
  1372.   this->next = after->next;
  1373.   after->next = this;
  1374.  
  1375.   if (verbose)
  1376.     printf ("%c - %s\n", old == 0 ? 'a' : 'r', this->info.name);
  1377.  
  1378.   return 1;
  1379. }
  1380.  
  1381. /* Apply a function to each of the specified members.
  1382. */
  1383.  
  1384. void
  1385. extract_members (function)
  1386.      void (*function) ();
  1387. {
  1388.   struct mapelt *map;
  1389.   FILE *arcstream;
  1390.   char **p;
  1391.  
  1392.   if (!files)
  1393.     {
  1394.       /* Handle case where we want to operate on every member.
  1395.      No need to make a map and search it for this.  */
  1396.       scan (function, 0);
  1397.       return;
  1398.     }
  1399.  
  1400.   arcstream = fopen (archive, "r");
  1401.   if (!arcstream)
  1402.     fatal ("failure opening archive %s for the second time", archive);
  1403.   map = make_map (0);
  1404.  
  1405.   for (p = files; *p; p++)
  1406.     {
  1407.       struct mapelt *this = find_mapelt (map, *p);
  1408.       if (!this) continue;
  1409.       function (this->info, arcstream);
  1410.     }
  1411.  
  1412.   fclose (arcstream);
  1413. }
  1414.  
  1415. /* Write the __.SYMDEF member from data in core.  OUTDESC and OUTNAME
  1416.    are descriptor and name of file to write to.  */
  1417.  
  1418. void
  1419. write_symdef_member (mapelt, map, outdesc, outname)
  1420.      struct mapelt *mapelt;
  1421.      struct mapelt *map;
  1422.      int outdesc;
  1423.      char *outname;
  1424. {
  1425.   struct ar_hdr header;
  1426.   int indesc;
  1427.   struct mapelt *mapptr;
  1428.   unsigned long int symdefs_size;
  1429.  
  1430.   if (mapelt->info.name == 0)
  1431.     /* This element was cancelled.  */
  1432.     return;
  1433.  
  1434.   header_from_map (&header, mapelt);
  1435.  
  1436.   bcopy (&header, &symdef_header, sizeof header);
  1437.  
  1438.   mywrite (outdesc, &header, sizeof (header), outname);
  1439.  
  1440.   /* Write the number of symdefs.  */
  1441.   symdefs_size = nsymdefs * sizeof (struct symdef);
  1442.   mywrite (outdesc, &symdefs_size, sizeof symdefs_size, outname);
  1443.  
  1444.   /* Write symdefs surviving from old archive.  */
  1445.   mywrite (outdesc, old_symdefs, num_old_symdefs * sizeof (struct symdef),
  1446.        outname);
  1447.  
  1448.   /* Write symdefs for new members.  */
  1449.   for (mapptr = map; mapptr != 0; mapptr = mapptr->next)
  1450.     if (mapptr->info.nsymdefs != 0)
  1451.       write (outdesc, mapptr->info.symdefs,
  1452.          mapptr->info.nsymdefs * sizeof (struct symdef));
  1453.  
  1454.   /* Write the string table size.  */
  1455.   mywrite (outdesc, &new_strings_size, sizeof new_strings_size, outname);
  1456.  
  1457.   /* Write the string table.  */
  1458.   mywrite (outdesc, new_strings, new_strings_size, outname);
  1459.  
  1460.   if (mapelt->info.size & 1)
  1461.     mywrite (outdesc, "", 1, outname);
  1462. }
  1463.  
  1464. void
  1465. read_old_symdefs (map, archive_indesc)
  1466.      struct mapelt *map;
  1467.      int archive_indesc;
  1468. {
  1469.   struct mapelt *mapelt;
  1470.   char *data;
  1471.   int val;
  1472.   int symdefs_size;
  1473.  
  1474.   mapelt = find_mapelt_noerror (map, "__.SYMDEF");
  1475.   if (!mapelt)
  1476.     abort ();            /* Only call here if an old one exists */
  1477.  
  1478.   data  = (char *) xmalloc (mapelt->info.size);
  1479.   lseek (archive_indesc, mapelt->info.data_offset, 0);
  1480.   val = read (archive_indesc, data, mapelt->info.size);
  1481.  
  1482.   symdefs_size = *(unsigned long int *) data;
  1483.   original_num_symdefs = symdefs_size / sizeof (struct symdef);
  1484.   old_symdefs = (struct symdef *) (data + sizeof (symdefs_size));
  1485.   old_strings = ((char *) (old_symdefs + original_num_symdefs)
  1486.          + sizeof (symdefs_size));
  1487.   old_strings_size
  1488.     = *(unsigned long int *) (old_symdefs + original_num_symdefs);
  1489. }
  1490.  
  1491. /* Read various information from the header of an object file.
  1492.    Return 0 for failure or 1 for success.  */
  1493.  
  1494. int
  1495. read_header_info (mapelt, desc, offset, syms_offset, syms_size, strs_offset, strs_size)
  1496.      struct mapelt *mapelt;
  1497.      int desc;
  1498.      long int offset;
  1499.      long int *syms_offset;
  1500.      unsigned int *syms_size;
  1501.      long int *strs_offset;
  1502.      unsigned int *strs_size;
  1503. {
  1504. #ifdef A_OUT
  1505.   {
  1506.     struct exec hdr;
  1507.  
  1508.     lseek (desc, offset, 0);
  1509. #ifdef HEADER_SEEK_FD
  1510.     HEADER_SEEK_FD (desc);
  1511. #endif
  1512.  
  1513.     if (read (desc, (char *) &hdr, sizeof hdr) == sizeof hdr && !N_BADMAG(hdr))
  1514.       {
  1515.     *syms_offset = N_SYMOFF (hdr);
  1516.     *syms_size = hdr.a_syms;
  1517.     *strs_offset = N_STROFF (hdr);
  1518.     lseek (desc, N_STROFF (hdr) + offset, 0);
  1519.     if (read (desc, (char *) strs_size, sizeof *strs_size) != sizeof *strs_size)
  1520.       {
  1521.         error_with_file ("failure reading string table size in ", mapelt);
  1522.         return 0;
  1523.       }
  1524.     return 1;
  1525.       }
  1526.   }
  1527. #endif
  1528.  
  1529. #ifdef MACH_O
  1530.   {
  1531.     struct mach_header mach_header;
  1532.     struct load_command *load_command;
  1533.     struct symtab_command *symtab_command;
  1534.     char *hdrbuf;
  1535.     int cmd, symtab_seen;
  1536.  
  1537.     lseek (desc, offset, 0);
  1538.     if (read (desc, (char *) &mach_header, sizeof mach_header) == sizeof mach_header
  1539.     && mach_header.magic == MH_MAGIC)
  1540.       {
  1541.     hdrbuf = xmalloc (mach_header.sizeofcmds);
  1542.     if (read (desc, hdrbuf, mach_header.sizeofcmds) != mach_header.sizeofcmds)
  1543.       {
  1544.         error_with_file ("failure reading load commands of ", mapelt);
  1545.         return 0;
  1546.       }
  1547.     load_command = (struct load_command *) hdrbuf;
  1548.     symtab_seen = 0;
  1549.     for (cmd = 0; cmd < mach_header.ncmds; ++cmd)
  1550.       {
  1551.         if (load_command->cmd == LC_SYMTAB)
  1552.           {
  1553.         symtab_seen = 1;
  1554.         symtab_command = (struct symtab_command *) load_command;
  1555.         *syms_offset = symtab_command->symoff;
  1556.         *syms_size = symtab_command->nsyms * sizeof (struct nlist);
  1557.         *strs_offset = symtab_command->stroff;
  1558.         *strs_size = symtab_command->strsize;
  1559.           }
  1560.         load_command = (struct load_command *) ((char *) load_command + load_command->cmdsize);
  1561.       }
  1562.     free (hdrbuf);
  1563.     if (!symtab_seen)
  1564.       *syms_offset = *syms_size = *strs_offset = *strs_size = 0;
  1565.     return 1;
  1566.       }
  1567.   }
  1568. #endif
  1569.  
  1570.   error_with_file ("bad format (not an object file) in ", mapelt);
  1571.   return 0;
  1572. }
  1573.  
  1574. /* Create the info.symdefs for a new member
  1575.    by reading the file it is coming from.  */
  1576.  
  1577. void
  1578. make_new_symdefs (mapelt, archive_indesc)
  1579.      struct mapelt *mapelt;
  1580.      int archive_indesc;
  1581. {
  1582.   int indesc;
  1583.   char *name = mapelt->info.name;
  1584.   long int syms_offset, strs_offset;
  1585.   unsigned int syms_size, strs_size;
  1586.   struct nlist *symbols;
  1587.   int symcount;
  1588.   char *strings;
  1589.   register unsigned int i;
  1590.   unsigned long int offset;
  1591.  
  1592.   if (name == 0)
  1593.     /* Deleted member.  */
  1594.     abort ();
  1595.  
  1596.   if (mapelt->info.offset != 0)
  1597.     {
  1598.       indesc = archive_indesc;
  1599.       lseek (indesc, mapelt->info.data_offset, 0);
  1600.       offset = mapelt->info.data_offset;
  1601.     }
  1602.   else
  1603.     {
  1604.       indesc = open (mapelt->info.name, 0, 0);
  1605.       if (indesc < 0)
  1606.     {
  1607.       perror_with_name (mapelt->info.name);
  1608.       return;
  1609.     }
  1610.       offset = 0;
  1611.     }
  1612.  
  1613.   if (!read_header_info (mapelt, indesc, offset, &syms_offset, &syms_size, &strs_offset, &strs_size))
  1614.     {
  1615.       if (mapelt->info.offset == 0)
  1616.     close (indesc);
  1617.       return;
  1618.     }
  1619.  
  1620.   /* Number of symbol entries in the file.  */
  1621.   symcount = syms_size / sizeof (struct nlist);
  1622.   /* Allocate temporary space for the symbol entries.  */
  1623.   symbols = (struct nlist *) alloca (syms_size);
  1624.   /* Read in the symbols.  */
  1625.   lseek (indesc, syms_offset + offset, 0);
  1626.   if (read (indesc, (char *) symbols, syms_size) != syms_size)
  1627.     {
  1628.       error_with_file ("premature end of file in symbols of ", mapelt);
  1629.       if (mapelt->info.offset == 0)
  1630.     (void) close (indesc);
  1631.       return;
  1632.     }
  1633.  
  1634.   /* The string table size includes the size word.  */
  1635.   if (strs_size < sizeof (strs_size))
  1636.     {
  1637.       error_with_file ("bad string table size in ", mapelt);
  1638.       if (mapelt->info.offset == 0)
  1639.     (void) close (indesc);
  1640.       return;
  1641.     }
  1642.   strs_size -= sizeof (strs_size);
  1643.  
  1644.   /* Allocate permanent space for the string table.  */
  1645.   strings = (char *) xmalloc (strs_size);
  1646.  
  1647.   /* Read in the strings.  */
  1648.   lseek (indesc, offset + strs_offset + sizeof strs_size, 0);
  1649.   if (read (indesc, strings, strs_size) != strs_size)
  1650.     {
  1651.       error_with_file ("premature end of file in strings of ", mapelt);
  1652.       if (mapelt->info.offset == 0)
  1653.     (void) close (indesc);
  1654.       return;
  1655.     }
  1656.  
  1657.   if (indesc != archive_indesc)
  1658.     (void) close (indesc);
  1659.  
  1660.   /* Discard the symbols we don't want to mention; compact the rest down.  */
  1661.   symcount = filter_symbols (symbols, symcount);
  1662.  
  1663.   mapelt->info.symdefs = (struct symdef *)
  1664.     xmalloc (symcount * sizeof (struct symdef));
  1665.   mapelt->info.nsymdefs = symcount;
  1666.   mapelt->info.string_size = 0;
  1667.  
  1668.   for (i = 0; i < symcount; ++i)
  1669.     {
  1670.       unsigned long int stroff = symbols[i].n_un.n_strx - sizeof (strs_size);
  1671.       char *symname = strings + stroff;
  1672.       if (stroff > strs_size)
  1673.     {
  1674.       char buf[100];
  1675.       sprintf (buf, "ridiculous string offset %lu in symbol %u of ",
  1676.            stroff + sizeof (strs_size), i);
  1677.       error_with_file (buf, mapelt);
  1678.       return;
  1679.     }
  1680.       mapelt->info.symdefs[i].s.name = symname;
  1681.       mapelt->info.string_size += strlen (symname) + 1;
  1682.     }
  1683. }
  1684.  
  1685. /* Choose which symbol entries to mention in __.SYMDEF;
  1686.    compact them downward to get rid of the rest.
  1687.    Return the number of symbols left.  */
  1688.  
  1689. int
  1690. filter_symbols (syms, symcount)
  1691.      struct nlist *syms;
  1692.      unsigned int symcount;
  1693. {
  1694.   struct nlist *from, *to;
  1695.   struct nlist *end = syms + symcount;
  1696.  
  1697.   for (to = from = syms; from < end; ++from)
  1698.     if ((from->n_type & N_EXT)
  1699.     && (from->n_type != N_EXT || from->n_value != 0))
  1700.       *to++ = *from;
  1701.  
  1702.   return to - syms;
  1703. }
  1704.  
  1705.  
  1706. /* Update the __.SYMDEF data before writing a new archive.  */
  1707.  
  1708. void
  1709. update_symdefs (map, archive_indesc)
  1710.      struct mapelt *map;
  1711.      int archive_indesc;
  1712. {
  1713.   struct mapelt *tail;
  1714.   int pos;
  1715.   register unsigned int i;
  1716.   unsigned int len;
  1717.   struct symdef *s;
  1718.   unsigned long int deleted_strings_size = 0;
  1719.  
  1720.   nsymdefs = original_num_symdefs;
  1721.   num_old_symdefs = original_num_symdefs;
  1722.   new_strings_size = old_strings_size;
  1723.  
  1724.   if (nsymdefs != 0)
  1725.     {
  1726.       /* We already had a __.SYMDEF member, so just update it.  */
  1727.  
  1728.       /* Mark as canceled any old symdefs for members being deleted.  */
  1729.  
  1730.       for (tail = map; tail != 0; tail = tail->next)
  1731.     {
  1732.       if (tail->info.name == 0)
  1733.         {
  1734.           /* Old member being deleted.  Delete its symdef entries too.  */
  1735.           for (i = 0; i < nsymdefs; i++)
  1736.         if (old_symdefs[i].offset == tail->info.offset)
  1737.           {
  1738.             old_symdefs[i].offset = 0;
  1739.             --nsymdefs;
  1740.             deleted_strings_size
  1741.               += strlen (old_strings
  1742.                  + old_symdefs[i].s.stringoffset) + 1;
  1743.           }
  1744.         }
  1745.     }
  1746.  
  1747.       /* Compactify old symdefs.  */
  1748.       {
  1749.     register unsigned int j = 0;
  1750.     for (i = 0; i < num_old_symdefs; ++i)
  1751.       {
  1752.         if (j != i)
  1753.           old_symdefs[j] = old_symdefs[i];
  1754.         if (old_symdefs[i].offset != 0)
  1755.           ++j;
  1756.       }
  1757.     num_old_symdefs -= i - j;
  1758.       }
  1759.  
  1760.       /* Create symdef data for any new members.  */
  1761.       for (tail = map; tail != 0; tail = tail->next)
  1762.     {
  1763.       if (tail->info.offset != 0
  1764.           || tail->info.name == 0
  1765.           || !strcmp (tail->info.name, "__.SYMDEF"))
  1766.         continue;
  1767.       make_new_symdefs (tail, archive_indesc);
  1768.       nsymdefs += tail->info.nsymdefs;
  1769.       new_strings_size += tail->info.string_size;
  1770.     }
  1771.     }
  1772.   else
  1773.     {
  1774.       /* Create symdef data for all existing members.  */
  1775.  
  1776.       for (tail = map; tail != 0; tail = tail->next)
  1777.     {
  1778.       if (tail->info.name == 0
  1779.           || !strcmp (tail->info.name, "__.SYMDEF"))
  1780.         continue;
  1781.       make_new_symdefs (tail, archive_indesc);
  1782.       nsymdefs += tail->info.nsymdefs;
  1783.       new_strings_size += tail->info.string_size;
  1784.     }
  1785.     }
  1786.  
  1787.   new_strings_size -= deleted_strings_size;
  1788.   old_strings_size -= deleted_strings_size;
  1789.  
  1790.   /* Now we know the size of __.SYMDEF,
  1791.      so assign the positions of all the members.  */
  1792.  
  1793.   tail = find_mapelt_noerror (map, "__.SYMDEF");
  1794.   tail->info.size = (sizeof (nsymdefs) + (nsymdefs * sizeof (struct symdef))
  1795.              + sizeof (new_strings_size) + new_strings_size);
  1796.   symdef_mapelt = tail;
  1797.  
  1798.   pos = SARMAG;
  1799.   for (tail = map; tail != 0; tail = tail->next)
  1800.     {
  1801.       if (tail->info.name == 0)
  1802.     /* Ignore deleted members.  */
  1803.     continue;
  1804.       tail->info.new_offset = pos;
  1805.       pos += sizeof (struct ar_hdr) + tail->info.size;
  1806.       if (tail->info.size & 1)
  1807.     ++pos;
  1808.     }
  1809.  
  1810.   /* Now update the offsets in the symdef data
  1811.      to be the new offsets rather than the old ones.  */
  1812.  
  1813.   for (tail = map; tail != 0; tail = tail->next)
  1814.     {
  1815.       if (tail->info.name == 0)
  1816.     continue;
  1817.       if (tail->info.symdefs == 0)
  1818.     /* Member without new symdef data.
  1819.        Check the old symdef data; it may be included there. */
  1820.     for (i = 0; i < num_old_symdefs; i++)
  1821.       {
  1822.         if (old_symdefs[i].offset == tail->info.offset)
  1823.           old_symdefs[i].offset = tail->info.new_offset;
  1824.       }
  1825.       else
  1826.     for (i = 0; i < tail->info.nsymdefs; i++)
  1827.       tail->info.symdefs[i].offset = tail->info.new_offset;
  1828.     }
  1829.  
  1830.   /* Generate new, combined string table and put each string's offset into the
  1831.      symdef that refers to it.  Note that old symdefs ref their strings by
  1832.      offsets into old_strings but new symdefs contain addresses of strings.  */
  1833.  
  1834.   new_strings = (char *) xmalloc (new_strings_size);
  1835.   pos = 0;
  1836.  
  1837.   /* Write the strings of the old symdefs and update the structures
  1838.      to contain indices into the string table instead of strings.  */
  1839.   for (i = 0; i < num_old_symdefs; i++)
  1840.     {
  1841.       strcpy (new_strings + pos, old_strings + old_symdefs[i].s.stringoffset);
  1842.       old_symdefs[i].s.stringoffset = pos;
  1843.       pos += strlen (new_strings + pos) + 1;
  1844.     }
  1845.   if (pos < old_strings_size)
  1846.     {
  1847.       unsigned int d = old_strings_size - pos;
  1848.       /* Correct the string table size.  */
  1849.       new_strings_size -= d;
  1850.       /* Correct the size of the `__.SYMDEF' member,
  1851.      since it contains the string table.  */
  1852.       symdef_mapelt->info.size -= d;
  1853.     }
  1854.   else if (pos > old_strings_size)
  1855.     fatal ("Old archive's string size was %u too small.",
  1856.        pos - old_strings_size);
  1857.  
  1858.   for (tail = map; tail != 0; tail = tail->next)
  1859.     if (tail->info.symdefs)
  1860.       {
  1861.     len = tail->info.nsymdefs;
  1862.     s = tail->info.symdefs;
  1863.  
  1864.     for (i = 0; i < len; i++)
  1865.       {
  1866.         strcpy (new_strings + pos, s[i].s.name);
  1867.         s[i].s.stringoffset = pos;
  1868.         pos += strlen (new_strings + pos) + 1;
  1869.       }
  1870.       }
  1871.   if (pos != new_strings_size)
  1872.     fatal ("internal error: inconsistency in new_strings_size", 0);
  1873. }
  1874.  
  1875. /* Print error message and usage message, and exit.  */
  1876.  
  1877. void
  1878. usage (s1, s2)
  1879.      char *s1, *s2;
  1880. {
  1881.   error (s1, s2);
  1882.   fprintf (stderr, "\
  1883. Usage: ar [d|m|p|q|r|t|x [[abi [position-name] [cilouv]] archive file...\n");
  1884.   exit (1);
  1885. }
  1886.  
  1887. /* Print error message and exit.  */
  1888.  
  1889. void
  1890. fatal (s1, s2)
  1891.      char *s1, *s2;
  1892. {
  1893.   error (s1, s2);
  1894.   exit (1);
  1895. }
  1896.  
  1897. /* Print error message.  `s1' is printf control string, the rest are args.  */
  1898.  
  1899. void
  1900. error (s1, s2, s3, s4, s5)
  1901.      char *s1, *s2, *s3, *s4, *s5;
  1902. {
  1903.   fprintf (stderr, "ar: ");
  1904.   fprintf (stderr, s1, s2, s3, s4, s5);
  1905.   fprintf (stderr, "\n");
  1906. }
  1907.  
  1908. void
  1909. error_with_file (string, mapelt)
  1910.      char *string;
  1911.      struct mapelt *mapelt;
  1912. {
  1913.   fprintf (stderr, "ar: ");
  1914.   fprintf (stderr, string);
  1915.   if (mapelt->info.offset != 0)
  1916.     fprintf (stderr, "%s(%s)", archive, mapelt->info.name);
  1917.   else
  1918.     fprintf (stderr, "%s", mapelt->info.name);
  1919.   fprintf (stderr, "\n");
  1920. }
  1921.  
  1922. void
  1923. perror_with_name (name)
  1924.      char *name;
  1925. {
  1926.   extern int errno, sys_nerr;
  1927.   extern char *sys_errlist[];
  1928.   char *s;
  1929.  
  1930.   if (errno < sys_nerr)
  1931.     s = concat ("", sys_errlist[errno], " for %s");
  1932.   else
  1933.     s = "unknown error for %s";
  1934.   error (s, name);
  1935. }
  1936.  
  1937. void
  1938. pfatal_with_name (name)
  1939.      char *name;
  1940. {
  1941.   extern int errno, sys_nerr;
  1942.   extern char *sys_errlist[];
  1943.   char *s;
  1944.  
  1945.   if (errno < sys_nerr)
  1946.     s = concat ("", sys_errlist[errno], " for %s");
  1947.   else
  1948.     s = "cannot open %s";
  1949.   fatal (s, name);
  1950. }
  1951.  
  1952. /* Return a newly-allocated string whose contents
  1953.    concatenate those of S1, S2, and S3.  */
  1954.  
  1955. char *
  1956. concat (s1, s2, s3)
  1957.      char *s1, *s2, *s3;
  1958. {
  1959.   int len1 = strlen (s1), len2 = strlen (s2), len3 = strlen (s3);
  1960.   char *result = (char *) xmalloc (len1 + len2 + len3 + 1);
  1961.  
  1962.   strcpy (result, s1);
  1963.   strcpy (result + len1, s2);
  1964.   strcpy (result + len1 + len2, s3);
  1965.   *(result + len1 + len2 + len3) = 0;
  1966.  
  1967.   return result;
  1968. }
  1969.  
  1970. /* Like malloc but get fatal error if memory is exhausted.  */
  1971.  
  1972. char *
  1973. xmalloc (size)
  1974.      unsigned int size;
  1975. {
  1976. #ifndef GNUDOS
  1977.   extern char *malloc ();
  1978. #endif
  1979.   char *result = malloc (size);
  1980.   if (result == 0)
  1981.     fatal ("virtual memory exhausted", 0);
  1982.   return result;
  1983. }
  1984.  
  1985. char *
  1986. xrealloc (ptr, size)
  1987.      char *ptr;
  1988.      unsigned int size;
  1989. {
  1990. #ifndef GNUDOS
  1991.   extern char *realloc ();
  1992. #endif
  1993.   char *result = realloc (ptr, size);
  1994.   if (result == 0)
  1995.     fatal ("virtual memory exhausted");
  1996.   return result;
  1997. }
  1998.  
  1999. #ifdef    USG
  2000. int
  2001. rename (from, to)
  2002.      char *from, *to;
  2003. {
  2004.   (void)unlink (to);
  2005.   if (link (from, to) < 0
  2006.       || unlink (from) < 0)
  2007.     return -1;
  2008.   else
  2009.     return 0;
  2010. }
  2011. #endif
  2012.